package com.luo.ai.mcp.server.config;

import com.luo.ai.mcp.server.tools.MyBaseTool;
import org.springframework.ai.tool.ToolCallbackProvider;
import org.springframework.ai.tool.method.MethodToolCallbackProvider;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

/**
 * 工具配置
 *
 * @author luohq
 * @date 2025-04-27 13:14
 */
@Configuration
public class ToolsConfig {

    /**
     * 注册工具
     *
     * @param toolArray 自定义工具实现类
     * @return 工具回调提供者
     */
    @Bean
    public ToolCallbackProvider myTools(MyBaseTool[] toolArray) {
        return MethodToolCallbackProvider.builder().toolObjects(toolArray).build();
    }
}
